package com.spzx.common.rabbit.service;

import com.spzx.common.core.utils.uuid.UUID;
import com.spzx.common.rabbit.entity.GuiguCorrelationData;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;

@Service
public class RabbitService {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 发送消息到RabbitMQ
     * 该方法实现了可靠消息发送机制，确保消息能够成功投递到RabbitMQ
     *
     * @param exchange   交换机名称，决定消息路由到哪个队列
     * @param routingKey 路由键，用于交换机将消息路由到具体队列
     * @param message    消息内容，可以是任意Java对象
     * @return 发送结果，true表示发送成功
     */
    public boolean sendMessage(String exchange, String routingKey, Object message) {
        // 创建消息关联数据对象，用于实现消息确认机制
        // GuiguCorrelationData继承自CorrelationData，用于跟踪消息状态
        GuiguCorrelationData correlationData = new GuiguCorrelationData();

        // 生成全局唯一的消息ID，用于标识每条消息
        // 使用UUID确保唯一性，并去除"-"字符使ID更简洁
        String uuid = "mq:" + UUID.randomUUID().toString().replaceAll("-", "");

        // 设置消息ID，用于后续的消息确认和追踪
        correlationData.setId(uuid);

        // 将要发送的消息内容保存到关联数据对象中
        correlationData.setMessage(message);

        // 保存交换机名称到关联数据对象中
        correlationData.setExchange(exchange);

        // 保存路由键到关联数据对象中
        correlationData.setRoutingKey(routingKey);

        // 将消息关联数据存储到Redis中，设置10分钟过期时间
        // 用于实现消息发送的可靠性保障，如果消息发送失败可以进行补偿
        redisTemplate.opsForValue().set(uuid, correlationData, 10, TimeUnit.MINUTES);

        // 使用RabbitTemplate发送消息到RabbitMQ
        // convertAndSend方法会自动将Object转换为消息体
        // correlationData参数用于消息确认机制，可以追踪消息状态
        rabbitTemplate.convertAndSend(exchange, routingKey, message, correlationData);

        // 返回发送成功标识
        return true;
    }


        /**
     * 发送延迟消息到RabbitMQ
     *
     * @param exchange   消息要发送到的交换机名称
     * @param routingKey 路由键
     * @param message    要发送的消息对象
     * @param delayTime  延迟时间(秒)
     * @return 发送成功返回true
     */
    public Boolean sendDealyMessage(String exchange, String routingKey, Object message, int delayTime) {
        //1.创建自定义相关消息对象-包含业务数据本身，交换器名称，路由键，队列类型，延迟时间,重试次数
        GuiguCorrelationData correlationData = new GuiguCorrelationData();
        String uuid = "mq:" + UUID.randomUUID().toString().replaceAll("-", "");
        correlationData.setId(uuid);
        correlationData.setMessage(message);
        correlationData.setExchange(exchange);
        correlationData.setRoutingKey(routingKey);
        correlationData.setDelay(true);
        correlationData.setDelayTime(delayTime);
        // 将消息关联数据存储到Redis中，设置10分钟过期时间
        redisTemplate.opsForValue().set(uuid, correlationData, 10, TimeUnit.MINUTES);

        // 发送延迟消息到RabbitMQ，设置消息的延迟时间
        rabbitTemplate.convertAndSend(exchange, routingKey, message, message1 -> {
            message1.getMessageProperties().setDelay(delayTime * 1000);
            return message1;
        }, correlationData);
        return true;
    }


}